from django.core.management.base import BaseCommand
from diet.models import FoodItem


class Command(BaseCommand):
    help = '加载示例食物数据'

    def handle(self, *args, **options):
        # 清空现有数据
        FoodItem.objects.all().delete()
        
        # 早餐食物
        breakfast_foods = [
            {
                'name': '燕麦粥',
                'category': 'breakfast',
                'calories': 150,
                'protein': 6.0,
                'carbs': 27.0,
                'fat': 3.0,
                'fiber': 4.0,
                'suitable_for': 'regular',
                'description': '营养丰富的燕麦粥，富含膳食纤维'
            },
            {
                'name': '全麦面包',
                'category': 'breakfast',
                'calories': 80,
                'protein': 4.0,
                'carbs': 15.0,
                'fat': 1.0,
                'fiber': 2.0,
                'suitable_for': 'regular',
                'description': '全麦面包，富含B族维生素'
            },
            {
                'name': '鸡蛋',
                'category': 'breakfast',
                'calories': 70,
                'protein': 6.0,
                'carbs': 1.0,
                'fat': 5.0,
                'fiber': 0.0,
                'suitable_for': 'regular',
                'description': '优质蛋白质来源'
            },
            {
                'name': '牛奶',
                'category': 'breakfast',
                'calories': 120,
                'protein': 8.0,
                'carbs': 12.0,
                'fat': 5.0,
                'fiber': 0.0,
                'suitable_for': 'regular',
                'description': '富含钙质的牛奶'
            },
            {
                'name': '豆浆',
                'category': 'breakfast',
                'calories': 80,
                'protein': 7.0,
                'carbs': 4.0,
                'fat': 4.0,
                'fiber': 1.0,
                'suitable_for': 'vegetarian',
                'description': '植物蛋白来源'
            }
        ]

        # 午餐食物
        lunch_foods = [
            {
                'name': '鸡胸肉',
                'category': 'lunch',
                'calories': 165,
                'protein': 31.0,
                'carbs': 0.0,
                'fat': 3.6,
                'fiber': 0.0,
                'suitable_for': 'regular',
                'description': '低脂高蛋白的鸡胸肉'
            },
            {
                'name': '三文鱼',
                'category': 'lunch',
                'calories': 208,
                'protein': 25.0,
                'carbs': 0.0,
                'fat': 12.0,
                'fiber': 0.0,
                'suitable_for': 'regular',
                'description': '富含Omega-3的三文鱼'
            },
            {
                'name': '糙米饭',
                'category': 'lunch',
                'calories': 110,
                'protein': 2.5,
                'carbs': 23.0,
                'fat': 0.9,
                'fiber': 1.8,
                'suitable_for': 'regular',
                'description': '富含膳食纤维的糙米'
            },
            {
                'name': '西兰花',
                'category': 'lunch',
                'calories': 55,
                'protein': 3.7,
                'carbs': 11.0,
                'fat': 0.6,
                'fiber': 5.2,
                'suitable_for': 'vegetarian',
                'description': '富含维生素C的西兰花'
            },
            {
                'name': '豆腐',
                'category': 'lunch',
                'calories': 76,
                'protein': 8.0,
                'carbs': 1.9,
                'fat': 4.8,
                'fiber': 0.3,
                'suitable_for': 'vegetarian',
                'description': '植物蛋白来源'
            }
        ]

        # 晚餐食物
        dinner_foods = [
            {
                'name': '瘦牛肉',
                'category': 'dinner',
                'calories': 250,
                'protein': 26.0,
                'carbs': 0.0,
                'fat': 15.0,
                'fiber': 0.0,
                'suitable_for': 'regular',
                'description': '富含铁质的瘦牛肉'
            },
            {
                'name': '蒸蛋羹',
                'category': 'dinner',
                'calories': 140,
                'protein': 12.0,
                'carbs': 2.0,
                'fat': 10.0,
                'fiber': 0.0,
                'suitable_for': 'regular',
                'description': '易消化的蒸蛋羹'
            },
            {
                'name': '红薯',
                'category': 'dinner',
                'calories': 103,
                'protein': 2.0,
                'carbs': 24.0,
                'fat': 0.2,
                'fiber': 3.8,
                'suitable_for': 'vegetarian',
                'description': '富含胡萝卜素的红薯'
            },
            {
                'name': '菠菜',
                'category': 'dinner',
                'calories': 23,
                'protein': 2.9,
                'carbs': 3.6,
                'fat': 0.4,
                'fiber': 2.2,
                'suitable_for': 'vegetarian',
                'description': '富含铁质的菠菜'
            },
            {
                'name': '蘑菇',
                'category': 'dinner',
                'calories': 22,
                'protein': 3.1,
                'carbs': 3.3,
                'fat': 0.3,
                'fiber': 1.0,
                'suitable_for': 'vegetarian',
                'description': '低热量的蘑菇'
            }
        ]

        # 零食
        snack_foods = [
            {
                'name': '苹果',
                'category': 'snack',
                'calories': 52,
                'protein': 0.3,
                'carbs': 14.0,
                'fat': 0.2,
                'fiber': 2.4,
                'suitable_for': 'vegetarian',
                'description': '富含果胶的苹果'
            },
            {
                'name': '坚果',
                'category': 'snack',
                'calories': 607,
                'protein': 20.0,
                'carbs': 21.0,
                'fat': 54.0,
                'fiber': 7.0,
                'suitable_for': 'vegetarian',
                'description': '富含健康脂肪的坚果'
            },
            {
                'name': '酸奶',
                'category': 'snack',
                'calories': 59,
                'protein': 10.0,
                'carbs': 3.6,
                'fat': 0.4,
                'fiber': 0.0,
                'suitable_for': 'regular',
                'description': '富含益生菌的酸奶'
            },
            {
                'name': '胡萝卜',
                'category': 'snack',
                'calories': 41,
                'protein': 0.9,
                'carbs': 10.0,
                'fat': 0.2,
                'fiber': 2.8,
                'suitable_for': 'vegetarian',
                'description': '富含胡萝卜素的胡萝卜'
            },
            {
                'name': '香蕉',
                'category': 'snack',
                'calories': 89,
                'protein': 1.1,
                'carbs': 23.0,
                'fat': 0.3,
                'fiber': 2.6,
                'suitable_for': 'vegetarian',
                'description': '富含钾的香蕉'
            }
        ]

        # 创建食物数据
        all_foods = breakfast_foods + lunch_foods + dinner_foods + snack_foods
        
        for food_data in all_foods:
            FoodItem.objects.create(**food_data)
        
        self.stdout.write(
            self.style.SUCCESS(f'成功创建了 {len(all_foods)} 种食物数据')
        )
